// Copyright 2021 by LunaSec (owned by Refinery Labs, Inc)
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
//
//go:build lambda

package main

import (
  "context"
  "errors"
  "github.com/aws/aws-lambda-go/events"
  "github.com/aws/aws-lambda-go/lambda"
  "github.com/lunasec-io/lunasec/lunadefend/go/pkg/tokenizerbackend"
)

var (
  logger, provider, gateways = tokenizerBackendDependencies()
  gatewayServer              = tokenizerbackend.NewApiGatewayServer(logger, provider, gateways)
)

func Handler(ctx context.Context, req events.APIGatewayProxyRequest) (events.APIGatewayProxyResponse, error) {
  defer func() {
    var err error
    if r := recover(); r != nil {
      switch x := r.(type) {
      case string:
        err = errors.New(x)
      case error:
        err = x
      default:
        err = errors.New("unknown panic")
      }
    }

    gateways.CW.PushMetrics()

    if err != nil {
      panic(err)
    }
  }()

  // If no name is provided in the HTTP request body, throw an error
  return gatewayServer.ProxyWithContext(ctx, req)
}

func main() {
  lambda.Start(Handler)
}
